import matplotlib.pyplot as plt


class ScatterDiagram(object):

    def create(self, path,red_data, orange_data, max_x):
        plt.plot(red_data[0], red_data[1], 'ro')
        plt.plot(orange_data[0], orange_data[1], 'yo')
        plt.axis([0, max_x, 40,100])
        plt.savefig(path)


if __name__ == '__main__':
    red_data = [[1, 2, 3, 50, 51, 61, 70], [70, 72, 81, 97, 99, 86, 90]]
    orange_data = [[3, 11, 19, 40, 45, 80, 90], [51, 53, 53, 53, 53, 53, 53]]
    max_x = 121
    chart = ScatterDiagram()
    chart.create(red_data, orange_data,max_x)
